tensor([[[0.2700, 0.2700],
         [0.2700, 0.2700]],

        [[0.4400, 0.4400],
         [0.4400, 0.4400]],

        [[0.2900, 0.2900],
         [0.2900, 0.2900]]], grad_fn=<RoundBackward1>)